package com.shuaigexue.aicodegenerate.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.mybatisflex.core.query.QueryWrapper;
import com.mybatisflex.spring.service.impl.ServiceImpl;
import com.shuaigexue.aicodegenerate.entity.User;
import com.shuaigexue.aicodegenerate.exception.BusinessException;
import com.shuaigexue.aicodegenerate.exception.ErrorCode;
import com.shuaigexue.aicodegenerate.mapper.UserMapper;
import com.shuaigexue.aicodegenerate.model.dto.user.UserQueryRequest;
import com.shuaigexue.aicodegenerate.model.enums.UserRoleEnum;
import com.shuaigexue.aicodegenerate.model.vo.LoginUserVO;
import com.shuaigexue.aicodegenerate.model.vo.UserVO;
import com.shuaigexue.aicodegenerate.service.UserService;
import jakarta.servlet.http.HttpServletRequest;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.function.BiConsumer;
import java.util.function.Consumer;
import java.util.stream.Collectors;

import static com.shuaigexue.aicodegenerate.constant.UserConstant.USER_LOGIN_STATE;

/**
 * 用户 服务层实现。
 *
 * @author <a href="https://github.com/liyupi">程序员鱼皮</a>
 */
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User>  implements UserService{

    @Override
    public long userRegister(String userAccount, String userPassword, String checkPassword) {
        // 1. 校验
        // 账户不能为空
        if(StrUtil.hasBlank(userAccount, userPassword, checkPassword)){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"参数为空");
        }
        // 账户不小于4位
        if(userAccount.length() < 4){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"用户账号过短");
        }
        // 密码不小于8位
        if(userPassword.length() < 8 || checkPassword.length() < 8){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"用户密码过短");
        }
        // 密码和校验密码相同
        if(!userPassword.equals(checkPassword)){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"两次输入的密码不一致");
        }
        // 账户不能包含特殊字符
        String validPattern = "^[a-zA-Z0-9_]+$";
        if(!userAccount.matches(validPattern)){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"用户账号不能包含特殊字符");
        }


        // 2.检查账户是否已存在
        // 创建查询条件构造器，用于构建数据库查询条件
        QueryWrapper queryWrapper = new QueryWrapper();
        // 添加查询条件：用户账户等于传入的userAccount参数
        queryWrapper.eq(User::getUserAccount, userAccount);
//        this.mapper.selectCountByQuery(queryWrapper);
        // 调用count方法，传入查询条件，返回结果
        long count = count(queryWrapper);
        if (count > 0) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR, "账号已存在");
        }

        // 3. 给账户密码加盐
        String encryptPassword = getEncryptPassword(userPassword);

        // 4. 插入数据,添加账户
        User user = new User();
        user.setUserAccount(userAccount);
        user.setUserPassword(encryptPassword);
        user.setUserName(userAccount);
        // 默认用户角色为普通用户,因为角色都是拥有权限的
        user.setUserRole(UserRoleEnum.USER.getValue());
        boolean saveResult = save(user);

        // 5. 返回插入数据id
        if(!saveResult){
            throw new BusinessException(ErrorCode.SYSTEM_ERROR, "注册失败，数据库错误");
        }
        return user.getId();
    }

    /**
     * 获取加密密码（加盐）
     * @param userPassword
     * @return
     */
    @Override
    public String getEncryptPassword(String userPassword) {
        final String solt = "shuaigexue";
        return DigestUtils.md5DigestAsHex((userPassword+solt).getBytes());
    }

    /**
     * 获取脱敏用户信息
     * @param user
     * @return
     */
    @Override
    public LoginUserVO getLoginUserVO(User user) {

        // 判断当前对象是否为空
        if(user ==  null){
            return null;
        }
        // 新建一个LoginUserVO对象
        LoginUserVO loginUserVO = new LoginUserVO();
        // user 不为空的话，将当前信息复制给loginUserVO对象
        BeanUtil.copyProperties(user,loginUserVO);
        return loginUserVO;
    }

    /**
     * 用户登录
     * @param userAccount 账户
     * @param userPassword 密码
     * @param request 请求
     * @return 脱敏用户信息
     */
    @Override
    public LoginUserVO userLogin(String userAccount, String userPassword, HttpServletRequest request) {
        // 1. 校验
        if(StrUtil.hasBlank(userAccount, userPassword)){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"参数为空");
        }
        if((userAccount.length())<4){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"账号错误");
        }
        if((userPassword.length())<8){
            throw new BusinessException(ErrorCode.PARAMS_ERROR,"密码错误");
        }
        // 2.密码加盐加密
        String encryptPassword = getEncryptPassword(userPassword);

        // 查询用户是否存在
        QueryWrapper queryWrapper = new QueryWrapper();
        queryWrapper.eq(User::getUserAccount, userAccount);
        queryWrapper.eq(User::getUserPassword,encryptPassword);
        // 根据查询条件查询用户
        User user = mapper.selectOneByQuery(queryWrapper);
        if(user == null){
            throw  new BusinessException(ErrorCode.PARAMS_ERROR,"用户不存在或密码错误");
        }
        // 3.记录用户登陆状态
        request.getSession().setAttribute(USER_LOGIN_STATE,user);
        return getLoginUserVO(user);
    }

    /**
     * 获取当前登录用户
     * @param request HTTP请求对象，用于获取客户端信息或存储会话数据
     * @return
     */
    @Override
    public User getLoginUser(HttpServletRequest request) {
        // 先判断是否登录
        Object userObj = request.getSession().getAttribute(USER_LOGIN_STATE);
        User currentUser = (User) userObj;

        return currentUser;
    }

    /**
     * 用户注销
     * @param request
     * @return
     */
    @Override
    public Boolean UserLogout(HttpServletRequest request) {
        Object userObj = request.getSession().getAttribute(USER_LOGIN_STATE);
        // 先判断当前用户是否登录
        if(userObj == null){
            throw  new BusinessException(ErrorCode.OPERATION_ERROR,"未登录");
        }
        // 是登陆状态的话，移除当前登录状态
        request.getSession().removeAttribute(USER_LOGIN_STATE);
        // 返回成功
        return true;
    }

    /**
     * 获取单个脱敏用户信息
     * @param user
     * @return
     */
    @Override
    public UserVO getUserVO(User user) {
        if (user == null) {
            return null;
        }
        UserVO userVO = new UserVO();
        BeanUtil.copyProperties(user, userVO);
        return userVO;
    }

    /**
     * 获取多个脱敏用户信息
     * @param userList
     * @return
     */
    @Override
    public List<UserVO> getUserVOList(List<User> userList) {
        if (CollUtil.isEmpty(userList)) {
            return new ArrayList<>();
        }
        return userList.stream().map(this::getUserVO).collect(Collectors.toList());
    }


    /**
     * 将请求转换为查询条件
     * @param userQueryRequest
     * @return
     */
    @Override
    public QueryWrapper getQueryWrapper(UserQueryRequest userQueryRequest) {
        if (userQueryRequest == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR, "请求参数为空");
        }

        Long id             = userQueryRequest.getId();
        String userAccount  = userQueryRequest.getUserAccount();
        String userName     = userQueryRequest.getUserName();
        String userProfile  = userQueryRequest.getUserProfile();
        String userRole     = userQueryRequest.getUserRole();
        String sortField    = userQueryRequest.getSortField();
        String sortOrder    = userQueryRequest.getSortOrder();

        // 创建查询包装器实例，用于构建动态查询条件
        QueryWrapper qw = QueryWrapper.create();

        // 定义一个BiConsumer函数式接口，用于处理模糊查询条件
        // 只有当值不为空时，才会添加对应的like查询条件
        BiConsumer<String, String> likeIfText =
                (col, val) -> {
                    // 使用Hutool工具类判断字符串是否不为空
                    if (StrUtil.isNotBlank(val)) {
                        // 添加模糊查询条件
                        qw.like(col, val);
                    }
                };

        // 定义一个BiConsumer函数式接口，用于处理相等查询条件
        // 只有当值不为空时，才会添加对应的eq查询条件
        BiConsumer<String, String> eqIfText =
                (col, val) -> {
                    // 使用Hutool工具类判断字符串是否不为空
                    if (StrUtil.isNotBlank(val)) {
                        // 添加相等查询条件
                        qw.eq(col, val);
                    }
                };

        // 定义一个Consumer函数式接口，用于处理ID的相等查询条件
        // 只有当ID大于0时，才会添加对应的eq查询条件
        Consumer<Long> eqIdIfValid =
                v -> {
                    // 判断ID是否有效（非空且大于0）
                    if (v != null && v > 0) {
                        // 添加ID相等查询条件
                        qw.eq("id", v);
                    }
                };

        // 动态添加查询条件（仅当参数有效时才追加）
        // 添加ID查询条件
        eqIdIfValid.accept(id);
        // 添加用户角色相等查询条件
        eqIfText.accept("userRole", userRole);
        // 添加用户账户模糊查询条件
        likeIfText.accept("userAccount", userAccount);
        // 添加用户名模糊查询条件
        likeIfText.accept("userName", userName);
        // 添加用户简介模糊查询条件
        likeIfText.accept("userProfile", userProfile);

        // 动态排序处理
        // 只有当排序字段不为空时，才会添加排序条件
        if (cn.hutool.core.util.StrUtil.isNotBlank(sortField)) {
            // 判断排序方式是否为升序（忽略大小写比较）
            boolean isAsc = "ascend".equalsIgnoreCase(sortOrder);
            // 添加排序条件
            qw.orderBy(sortField, isAsc);
        }

        // 返回构建好的查询包装器
        return qw;
    }
//    @Override
//    public QueryWrapper getQueryWrapper(UserQueryRequest userQueryRequest) {
//        if (userQueryRequest == null) {
//            throw new BusinessException(ErrorCode.PARAMS_ERROR, "请求参数为空");
//        }
//        Long id = userQueryRequest.getId();
//        String userAccount = userQueryRequest.getUserAccount();
//        String userName = userQueryRequest.getUserName();
//        String userProfile = userQueryRequest.getUserProfile();
//        String userRole = userQueryRequest.getUserRole();
//        String sortField = userQueryRequest.getSortField();
//        String sortOrder = userQueryRequest.getSortOrder();
//        return QueryWrapper.create()
//                .eq( UserVO::getId, id)
//                .eq( UserVO::getUserRole, userRole,StrUtil.isNotBlank(userRole))
//                .like( UserVO::getUserAccount, userAccount, StrUtil.isNotBlank(userAccount))
//                .like( UserVO::getUserName, userName, StrUtil.isNotBlank(userName))
//                .like( UserVO::getUserProfile, userProfile, StrUtil.isNotBlank(userProfile))
//                .orderBy(sortField, "ascend".equals(sortOrder));
//    }



}
